package com.runfast.waimai.utils;


import java.awt.Color;
import java.awt.Graphics;
import java.awt.Image;
import java.awt.image.BufferedImage;
import java.awt.image.ColorModel;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Arrays;
import javax.imageio.IIOImage;
import javax.imageio.ImageIO;
import javax.imageio.ImageTypeSpecifier;
import javax.imageio.ImageWriteParam;
import javax.imageio.ImageWriter;



public class CompressPic {
	public static boolean compress(String srcPath,File descPath,float quality){
		boolean flag = false;
		File file = null;
		BufferedImage src = null;
		FileOutputStream out = null;
		ImageWriter imageWriter;
		ImageWriteParam imageWriteParam;

		//指定格式为jpg
		imageWriter = ImageIO.getImageWritersByFormatName("jpg").next();

		imageWriteParam = new javax.imageio.plugins.jpeg.JPEGImageWriteParam(null);
		// 要使用压缩，必须指定压缩方式为MODE_EXPLICIT
		imageWriteParam.setCompressionMode(ImageWriteParam.MODE_EXPLICIT);
		// 这里指定压缩的程度，参数qality是取值0~1范围内，
		imageWriteParam.setCompressionQuality(quality);
		imageWriteParam.setProgressiveMode(ImageWriteParam.MODE_DISABLED);
		ColorModel colorModel = ColorModel.getRGBdefault();
		imageWriteParam.setDestinationType(new ImageTypeSpecifier(colorModel, colorModel.createCompatibleSampleModel(16, 16)));
		if(!ComUtil.isEmpty(srcPath)){
			file = new File(srcPath);
			try {
				src = ImageIO.read(file);
				out = new FileOutputStream(descPath);
				imageWriter.reset();
				// 必须先指定 out值，才能调用write方法, ImageOutputStream可以通过任何 OutputStream构造
				imageWriter.setOutput(ImageIO.createImageOutputStream(out));
				// 调用write方法，就可以向输入流写图片
				imageWriter.write(null, new IIOImage(src, null, null), imageWriteParam);
				out.flush();
				out.close();
				flag = true;
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return flag;
	}
	/**
	 * <p>Title: thumbnailImage</p>
	 * <p>Description: 根据图片路径生成缩略图 </p>
	 * @param //imagePath    原图片路径
	 * @param w            缩略图宽
	 * @param h            缩略图高
	 * @param prevfix    生成缩略图的前缀
	 * @param force        是否强制按照宽高生成缩略图(如果为false，则生成最佳比例缩略图)
	 */
	public static void thumbnailImage(File imgFile, int w, int h, String prevfix, boolean force){
		if(imgFile.exists()){
			try {
				// ImageIO 支持的图片类型 : [BMP, bmp, jpg, JPG, wbmp, jpeg, png, PNG, JPEG, WBMP, GIF, gif]
				String types = Arrays.toString(ImageIO.getReaderFormatNames());
				String suffix = null;
				// 获取图片后缀
				if(imgFile.getName().indexOf(".") > -1) {
					suffix = imgFile.getName().substring(imgFile.getName().lastIndexOf(".") + 1);
				}// 类型和图片后缀全部小写，然后判断后缀是否合法
				if(suffix == null || types.toLowerCase().indexOf(suffix.toLowerCase()) < 0){
					System.out.println("Sorry, the image suffix is illegal. the standard image suffix is "+types+"." );
					return ;
				}
				System.out.println("target image's size, width:"+w+", height:"+h+".");
				Image img = ImageIO.read(imgFile);
				if(!force){
					// 根据原图与要求的缩略图比例，找到最合适的缩略图比例
					int width = img.getWidth(null);
					int height = img.getHeight(null);
					if((width*1.0)/w < (height*1.0)/h){
						if(width > w){
							h = Integer.parseInt(new java.text.DecimalFormat("0").format(height * w/(width*1.0)));
							System.out.println("change image's height, width:"+w+", height:"+h+".");
						}
					} else {
						if(height > h){
							w = Integer.parseInt(new java.text.DecimalFormat("0").format(width * h/(height*1.0)));
							System.out.println("change image's width, width:"+w+", height:"+h+".");
						}
					}
				}
				BufferedImage bi = new BufferedImage(w, h, BufferedImage.TYPE_INT_RGB);
				Graphics g = bi.getGraphics();
				g.drawImage(img, 0, 0, w, h, Color.LIGHT_GRAY, null);
				g.dispose();
				String p = imgFile.getPath();
				// 将图片保存在原目录并加上前缀
				ImageIO.write(bi, suffix, new File(p.substring(0,p.lastIndexOf(File.separator)) + File.separator + prevfix +imgFile.getName()));
			} catch (IOException e) {
				System.out.println("generate thumbnail image failed:"+e.getMessage());
			}
		}else{
			System.out.println("the image is not exist.");
		}
	}

}
